import os
PATH = os.getcwd()
images_path = os.path.join(PATH, 'images')
images = os.listdir(images_path)
with open('train.txt', 'w') as f:
    for image in images:
        f.write(os.path.join(images_path, image))
        f.write('\n')
